python main.py \
  --run_name "test_gpt3" \
  --root_dir "root" \
  --dataset_path ./benchmarks/mbpp-py.jsonl \
  --strategy "mcts" \
  --language "py" \
  --model "gpt-4o-mini" \
  --pass_at_k "1" \
  --max_iters "4" \
  --expansion_factor "3" \
  --number_of_tests "4" \
  --verbose \
  --n_generate_sample 5 \
  --save_path trajectories \
